/***
    mockcpp is a generic C/C++ mock framework.
    Copyright (C) <2010>  <Darwin Yuan: darwin.yuan@gmail.com>
	                      <Chen Guodong: sinojelly@gmail.com>

    This program is free software: you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
    the Free Software Foundation, either version 3 of the License, or
    (at your option) any later version.

    This program is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    GNU General Public License for more details.

    You should have received a copy of the GNU General Public License
    along with this program.  If not, see <http://www.gnu.org/licenses/>.
***/

#ifndef __MOCKCPP_BLOCKALLOCATOR_H__
#define __MOCKCPP_BLOCKALLOCATOR_H__

#include <mockcpp/PageAllocator.h>

MOCKCPP_NS_START


struct BlockAllocator : public MemAllocator
{    
    BlockAllocator(unsigned int blockSize, unsigned int blockNum);
	BlockAllocator(unsigned int blockSize, MOCKCPP_NS::PageAllocator *poolAllocator);
    ~BlockAllocator();
    void* alloc(size_t size);
    void free(void* ptr);
    bool freeable(void* ptr);
	
private:
    struct Block 
    {
        Block* next;        
    };

private:
    bool isBadBlock(void* ptr) const;
    bool misaligned(void* ptr) const;
    bool outOfScope(void* ptr) const;
    bool hasBeenFreed(void* ptr) const;

private:
    unsigned int sizeOfBlock;
    unsigned int numOfBlocks;
	MOCKCPP_NS::MemAllocator *pageAllocator;
    void* buffer;
    Block* headOfFreeList;	
	unsigned int sizeOfPool;
	unsigned int numOfPool;
};

MOCKCPP_NS_END

#endif


